import os										
from langchain_community.graphs import Neo4jGraph


def conf_neo4j():
    os.environ["NEO4J_URI"] = "bolt://localhost:7687"
    os.environ["NEO4J_USERNAME"] = "neo4j"
    os.environ["NEO4J_PASSWORD"] = "970412@wcx.com"


def get_neo4j_graph():
    conf_neo4j() 
    graph = Neo4jGraph()
    print("Neo4j config successfully.")
    return graph 


if __name__ == "__main__":
    get_neo4j_graph()
